from ultralytics import YOLO
import cv2

# 加载模型(yolov8n.pt为轻量版, yolov8x.pt为高精度版)
model = YOLO('yolov8n.pt')

# 检测图片
image_path = 'test.jpg'
results = model(image_path)

# 显示结果（标注后的图片）
for r in results:
    im_array = r.plot()  # 生成带标注的图像
    cv2.imshow('YOLO Detection', im_array)  # 显示窗口
    cv2.waitKey(0)  # 按任意键关闭窗口

cv2.destroyAllWindows()